# 用于绘制带有子图的图像。来源：https://www.jianshu.com/p/de223a79217a，本人整理。
# !/usr/bin/env python
# !encoding=utf-8

import matplotlib.pyplot as plt
import numpy as np


def f(t):
    return np.exp(-t) * np.cos(2 * np.pi * t)


t1 = np.arange(0, 5, 0.1)
t2 = np.arange(0, 5, 0.02)

plt.figure()
plt.subplot(221)
plt.plot(t1, f(t1), color='b', marker='o', label='decaying')
plt.legend(loc='upper right')  # 将图例放在右上角。若输入lower right，代表右下角，以此类推。
plt.subplot(222)
plt.plot(t2, np.cos(2 * np.pi * t2), 'r--', label='sinsidual')
plt.legend(loc='upper right')  # 将图例放在右上角。若输入lower right，代表右下角，以此类推。
plt.subplot(212)
plt.plot([1, 2, 3, 4], [1, 4, 9, 16], label='sqrt')
plt.legend(loc='upper right')  # 将图例放在右上角。若输入lower right，代表右下角，以此类推。
plt.show()
